#pragma once

#include <string>
#include <vector>
#include "db_handler.h"

namespace TakeAwayPlatform
{
    class MerchantModule
    {
    public:
        MerchantModule(DatabaseHandler& dbHandler);

        bool addDish(int merchant_id, const std::string& dish_name, double price, const std::string& description, const std::string& category, int stock);
        bool updateDish(int dish_id, const std::string& dish_name, double price, const std::string& description, const std::string& category, int stock);
        bool deleteDish(int dish_id);
        bool viewDishes(int merchant_id, std::vector<std::string>& dish_names);

    private:
        DatabaseHandler& dbHandler;
    };
}
